package com.da.result.model;

import java.util.ArrayList;
import java.util.List;

/**
 * 保存返回编码结果的模型
 *
 * @author da
 * @time 2024/7/12 下午 4:00
 */
public class Embedding {

    /**
     * 词向量列表
     */
    private final List<Double> embedding = new ArrayList<>();

    public Embedding() {}

    public Embedding(List<Double> embedding) {
        this.embedding.addAll(embedding);
    }

    /**
     * 添加词向量
     *
     * @param d 词向量
     */
    public void add(double d) {
        embedding.add(d);
    }

    /**
     * 获取词向量列表
     *
     * @return 词向量列表
     */
    public List<Double> getEmbedding() {
        return embedding;
    }

    @Override
    public String toString() {
        return "{" + "\"embedding\":" + embedding + '}';
    }
}
